This is the accompanying repo to the paper "Many processors, little time: MCMC for partitions via optimal transport couplings", to appear in AISTATS 2022.

To install necessary Python packages, make the environment specified by the environment.yml.

Generate figures in main text

Results for gene dataset

Run new experiments

Run a minimal example

The list of examples include

• Mixture model on gene dataset
  • Gibbs sampler
  • Jain-Neal split merge
• Graph coloring on k-regular graph

Please contact tdn@mit.edu if you have any questions!